package com.xuanzheng.company.dao;

import com.xuanzheng.company.entity.AIChatRecord;
import com.xuanzheng.company.entity.AIUsageLimit;

import java.util.List;

/**
 * AI聊天数据访问接口
 */
public interface AIChatDao {
    
    /**
     * 保存聊天记录
     */
    boolean saveChatRecord(AIChatRecord record);
    
    /**
     * 更新聊天记录
     */
    boolean updateChatRecord(AIChatRecord record);
    
    /**
     * 根据用户ID获取聊天记录
     */
    List<AIChatRecord> findChatRecordsByUserId(String userId, int limit);
    
    /**
     * 获取用户使用限制
     */
    AIUsageLimit findUsageLimitByUserId(String userId);
    
    /**
     * 创建用户使用限制
     */
    boolean createUsageLimit(AIUsageLimit limit);
    
    /**
     * 更新用户使用限制
     */
    boolean updateUsageLimit(AIUsageLimit limit);
    
    /**
     * 增加用户今日使用次数
     */
    boolean incrementUsageCount(String userId);
    
    /**
     * 重置所有用户的每日使用次数
     */
    boolean resetDailyUsage();
}